#include "napi/native_api.h"
#include "native_hook.h"
#include "hilog/log.h"
#define LOG_TAG "testTag"

static native_hooker __hooker;

static napi_value Add(napi_env env, napi_callback_info info) {

    size_t argc = 2;
    napi_value args[2] = {nullptr};

    napi_get_cb_info(env, info, &argc, args, nullptr, nullptr);

    napi_valuetype valuetype0;
    napi_typeof(env, args[0], &valuetype0);

    napi_valuetype valuetype1;
    napi_typeof(env, args[1], &valuetype1);

    double value0;
    napi_get_value_double(env, args[0], &value0);

    double value1;
    napi_get_value_double(env, args[1], &value1);

    napi_value sum;
    napi_create_double(env, value0 + value1, &sum);

    return sum;
}



// hook点1：系统库中NativeEngine类的Init方法
void (*__old_impl_init)(void *);

void __nativehook_impl_init(void *_this) {
    // 会进入此方法
    OH_LOG_ERROR(LOG_APP, "Enter: __nativehook_impl_init, line: %{public}d", __LINE__);
    __old_impl_init(_this);
}


EXTERN_C_START
static napi_value Init(napi_env env, napi_value exports) {
    // 初始化钩子
    __hooker.phrase_proc_maps();
    // 方式一：注册钩子，搜索全部so
//    __hooker.hook_all_modules("_ZN12NativeEngine4InitEv",
//                              (void *)__nativehook_impl_init,
//                              (void **)&__old_impl_init,
//                              true);
    // 方式二：注册钩子，指定so名称，效率较高
    __hooker.hook_module("/system/lib64/platformsdk/libace_napi.z.so",
                         "_ZN12NativeEngine4InitEv",
                         (void *)__nativehook_impl_init,
                         (void **)&__old_impl_init,
                         true);


    napi_property_descriptor desc[] = {{"add", nullptr, Add, nullptr, nullptr, nullptr, napi_default, nullptr}};
    napi_define_properties(env, exports, sizeof(desc) / sizeof(desc[0]), desc);
    return exports;
}
EXTERN_C_END

static napi_module demoModule = {
    .nm_version = 1,
    .nm_flags = 0,
    .nm_filename = nullptr,
    .nm_register_func = Init,
    .nm_modname = "entry",
    .nm_priv = ((void *)0),
    .reserved = {0},
};

extern "C" __attribute__((constructor)) void RegisterEntryModule(void) { napi_module_register(&demoModule); }
